implementation of neural network in python